"""
使用 LangServe 部署高级聊天机器人的 Web API
"""

import os
from fastapi import FastAPI
from langserve import add_routes
from dotenv import load_dotenv
from chatbot import AdvancedChatbot

# 加载环境变量
load_dotenv()

# 创建 FastAPI 应用
app = FastAPI(
    title="高级聊天机器人 API",
    version="1.0",
    description="基于 LangGraph 的高级聊天机器人，支持 Web 搜索、对话状态保持和人工审核"
)

# 创建聊天机器人实例
chatbot = AdvancedChatbot(debug=False)

# 添加 LangServe 路由
add_routes(
    app,
    chatbot.workflow,
    path="/chat",
    config_keys=["configurable"],
    playground_type="chat"
)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000) 